from selenium import webdriver
from selenium.webdriver.common.by import By

from webselenium.po.basepage import BasePage
from webselenium.po.contact_page import ContactPage

#首页
class MainPage(BasePage):
    _CONTACTS = (By.ID, "menu_contacts")

    #首页对应的操作为点击通讯录，所以封装点击通讯录
    def goto_contact(self):
        #优化重复代码，将浏览器复用在basepage中定义
        # opt = webdriver.ChromeOptions()
        # opt.debugger_address = "127.0.0.1:9222"
        # self.driver = webdriver.Chrome(options=opt)
        # # self.driver = webdriver.Chrome()
        # self.driver.implicitly_wait(5)
        # self.driver.get("https://work.weixin.qq.com/wework_admin/frame")

        # self.driver.find_element_by_id("menu_contacts").click()
        # 优化代码，使用在basepage中封装的方法
        self.find_and_click(By.ID, "menu_contacts")
        self.find_and_click(*self._CONTACTS)


        #完成相关后，需要跳转到通讯录页面，即返回到通讯录页面对象中
        # self.driver是从basepage中继承过来
        return ContactPage(self.driver)



